package com.gewei.utils;

import com.google.gson.Gson;
import com.qiniu.common.QiniuException;
import com.qiniu.common.Zone;
import com.qiniu.http.Response;
import com.qiniu.storage.BucketManager;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.UploadManager;
import com.qiniu.storage.model.DefaultPutRet;
import com.qiniu.util.Auth;

/**
 * 七牛云工具类
 * 使用七牛云提供的 SDK 实现将本地图片上传到七牛云服务器
 */
public class QiniuUtils {
    /**
     * ...生成上传凭证，然后准备上传
     */
    private static String accessKey = "0x5iKfEsqnmFBwOTsIXpwwDJ-i7P5VAhZd_9kTB6";
    private static String secretKey = "Cskt9Gb2S3PM_c7AAK_lD82ea11olW0QDSHGRyrs";
    private static String bucket = "ithealth-gewei";

    /**
     * 上传文件
     *
     * @param filePath
     * @param fileName
     */
    public static void upload2Qiniu(String filePath, String fileName) {
        //  构造一个带指定 Zone 对象的配置类
        Configuration cfg = new Configuration(Zone.zone2());
        UploadManager uploadManager = new UploadManager(cfg);
        Auth auth = Auth.create(accessKey, secretKey);
        String upToken = auth.uploadToken(bucket);
        try {
            Response response = uploadManager.put(filePath, fileName, upToken);
            //  解析上传成功的结果
            DefaultPutRet putRet = new Gson().fromJson(response.bodyString(), DefaultPutRet.class);
        } catch (QiniuException ex) {
            Response r = ex.response;
            try {
                System.err.println(r.bodyString());
            } catch (QiniuException ex2) {
                //  ignore
            }
        }
    }

    /**
     * 上传文件
     *
     * @param bytes
     * @param fileName
     */
    public static void upload2Qiniu(byte[] bytes, String fileName) {
        //  构造一个带指定 Zone 对象的配置类
        Configuration cfg = new Configuration(Zone.zone0());
        //  ...其他参数参考类注释
        UploadManager uploadManager = new UploadManager(cfg);

        //  默认不指定 key 的情况下，以文件内容的 hash 值作为文件名
        String key = fileName;
        //  鉴权
        Auth auth = Auth.create(accessKey, secretKey);
        //  上传令牌
        String upToken = auth.uploadToken(bucket);
        try {
            Response response = uploadManager.put(bytes, key, upToken);
            //  解析上传成功的结果
            DefaultPutRet putRet = new Gson().fromJson(response.bodyString(), DefaultPutRet.class);
            System.out.println(putRet.key);
            System.out.println(putRet.hash);
        } catch (QiniuException ex) {
            Response r = ex.response;
            System.err.println(r.toString());
            try {
                System.err.println(r.bodyString());
            } catch (QiniuException ex2) {
                //ignore
            }
        }
    }

    /**
     * 删除七牛云服务器中的文件
     *
     * @param fileName
     */
    public static void deleteFileFromQiniu(String fileName) {
        //  构造一个带指定 Zone 对象的配置类
        Configuration cfg = new Configuration(Zone.zone0());
        String key = fileName;
        Auth auth = Auth.create(accessKey, secretKey);
        BucketManager bucketManager = new BucketManager(auth, cfg);
        try {
            bucketManager.delete(bucket, key);
            System.out.println("删除成功");
        } catch (QiniuException ex) {
            //  如果遇到异常，说明删除失败
            System.err.println(ex.code());
            System.err.println(ex.response.toString());
        }
    }
}